Two-Stage Reinforcement Learning for Sound and Adversarial Test Generation in Code LLMs
两阶段强化学习生成可靠对抗测试用例提升代码 LLM
🧭 代码 LLM 测试生成 · 对抗 RL
Nanyang Technological University, Singapore · Skywork AI
提出两阶段 RL 测试生成框架 TCS,先学可靠测试再对抗生成反例,提升代码生成 pass@1 与答案选择
📌 工作参考价值(4 点)
- 若业务任务具备可自动验证的 reward(如商品属性合法性、SQL 可执行、Promo 规则检查、点击/转化模拟器),可借鉴 TCS 两阶段生成器:Stage 1 贴近参考解/规则生成合法样本,Stage 2 针对线上模型常见失败模式生成反例,用于训练更鲁棒的排序/生成模型或筛选候选。
- 将测试生成器作为可学习的 critic:在电商 Agent 或推荐 query/文案生成中,训练一个判别器为候选输出打分,替代或辅助人工评审;用当前策略的失败样本构造滚动缓冲,持续挖掘 hard negatives,提升候选筛选的区分能力。
- 推理时选择:生成多个候选推荐理由/搜索 query/广告标题,用学到的检查器自动过滤不合格输出,降低低质量内容上线风险;类似无参考评估,适合在线管道。
- 注意领域差异:代码执行反馈明确且低成本,电商反馈(点击/转化)噪声大、延迟高,需用离线模拟器或业务规则先构造可信 reward,不宜直接照搬对抗 RL 的完全在线迭代。